#!/usr/bin/python
# -*- coding: UTF-8 -*-
"""
@author:吹神
@file:json_path.py
@time:2021/04/07
"""

from jsonpath import jsonpath
from config.settings import log


class JsonPath:

    def __init__(self, s, e):
        self.data = jsonpath(s, e)
        if not self.data:
            log.error(f"s: {s} \n e: {e}")
            raise ValueError("jsonpath表达式错误")

    def all(self):
        # 获取所有数据
        return self.data

    def len(self):
        # 获取数据长度
        return len(self.data)

    def one(self):
        # 获取第一个数据
        return self.data[0]
